;; Licensed to the Apache Software Foundation (ASF) under one
;; or more contributor license agreements.  See the NOTICE file
;; distributed with this work for additional information
;; regarding copyright ownership.  The ASF licenses this file
;; to you under the Apache License, Version 2.0 (the
;; "License"); you may not use this file except in compliance
;; with the License.  You may obtain a copy of the License at
;;
;; http://www.apache.org/licenses/LICENSE-2.0
;;
;; Unless required by applicable law or agreed to in writing, software
;; distributed under the License is distributed on an "AS IS" BASIS,
;; WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
;; See the License for the specific language governing permissions and
;; limitations under the License.
(ns backtype.storm.messaging.loader
  (:import [backtype.storm.messaging IConnection DeserializingConnectionCallback])
  (:require [backtype.storm.messaging [local :as local]]))

(defn mk-local-context []
  (local/mk-context))

(defn- mk-connection-callback
  "make an IConnectionCallback"
  [transfer-local-fn storm-conf worker-context]
  (DeserializingConnectionCallback. storm-conf
                                    worker-context
                                    (fn [batch]
                                      (transfer-local-fn batch))))

(defn register-callback
  "register the local-transfer-fn with the server"
  [transfer-local-fn ^IConnection socket storm-conf worker-context]
  (.registerRecv socket (mk-connection-callback transfer-local-fn storm-conf worker-context)))
